# config.py
import os
from dotenv import load_dotenv

load_dotenv() # 加载环境变量

class envConfig:
    # Ollama 配置（保持原样）
    OLLAMA_EMBEDDING = "nomic-embed-text"  # 嵌入模型名称
    OLLAMA_URL = "http://localhost:11434"  # Ollama 服务地址

    # Milvus Docker 配置
    MILVUS_HOST = "localhost"  # 根据您的Milvus服务器地址修改
    MILVUS_PORT = "19530"  # 默认端口
    MILVUS_LOCAL_URI = "http://localhost:19530"  # Docker 映射到本机的 Milvus 地址
    MILVUS_USER = ""  # 如果未启用认证则留空
    MILVUS_PASSWORD = ""  # 如果未启用认证则留空

    # 云端备份配置（根据实际情况填写）
    # BACKUP_MILVUS_URI = "your_cloud_uri"
    # BACKUP_MILVUS_USERNAME = "your_cloud_user"
    # BACKUP_MILVUS_PASSWORD = "your_cloud_password"

    #正式环境中加个认证
    # MILVUS_USER = "root"  # 默认管理员账号
    # MILVUS_PASSWORD = "Milvus@2025"  # 自定义强密码

    # config.py 中添加
    class envConfig:
        DEEPSEEK_API_KEY = "sk-c8271445bd834602a7f6960000000000"  # 替换为你的 API Key
        DEEPSEEK_BASE_URL = "https://api.deepseek.com/v1"  # API 地址
        APP_NAME = "InsuranceQA" #deepseek0脚本新增
        # Milvus配置
        MILVUS_HOST = os.getenv("MILVUS_HOST", "localhost")
        MILVUS_PORT = os.getenv("MILVUS_PORT", "19530")
        MILVUS_USER = os.getenv("MILVUS_USER", "")
        MILVUS_PASSWORD = os.getenv("MILVUS_PASSWORD", "")
        # Ollama配置
        OLLAMA_URL = os.getenv("OLLAMA_URL", "http://localhost:11434")


class config:
    APP_NAME = "MyLangChainApp"  # 应用名称（用于集合命名）
    # Docker 中 Milvus 的持久化存储路径（可选）
    DOCKER_VOLUME_PATH = "/var/lib/milvus/data"  # 数据持久化目录

